from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
import time

# 初始化Chrome浏览器
driver = webdriver.Chrome()

# 打开颐和园官方售票网站
driver.get("https://ticket.summerpalace-china.com/#/")

# 等待页面加载
time.sleep(5)

try:
    # 查找登录按钮并点击（假设登录按钮的ID为'login-button'）
    login_button = driver.find_element(By.ID, 'login-button')
    login_button.click()

    # 等待登录页面加载
    time.sleep(3)

    # 输入用户名和密码（假设用户名和密码输入框的ID分别为'username'和'password'）
    username_input = driver.find_element(By.ID, 'username')
    username_input.send_keys('your_username')

    password_input = driver.find_element(By.ID, 'password')
    password_input.send_keys('your_password')

    # 提交登录表单
    password_input.send_keys(Keys.RETURN)

    # 等待登录完成
    time.sleep(5)

    # 查找购票按钮并点击（假设购票按钮的ID为'buy-ticket-button'）
    buy_ticket_button = driver.find_element(By.ID, 'buy-ticket-button')
    buy_ticket_button.click()

    # 等待购票页面加载
    time.sleep(3)

    # 选择票种并提交订单（假设票种选择框的ID为'ticket-type'）
    ticket_type_select = driver.find_element(By.ID, 'ticket-type')
    ticket_type_select.send_keys('desired_ticket_type')

    # 提交订单（假设提交订单按钮的ID为'submit-order-button'）
    submit_order_button = driver.find_element(By.ID, 'submit-order-button')
    submit_order_button.click()

    # 等待订单提交完成
    time.sleep(5)

    print("购票成功！")

except Exception as e:
    print(f"购票失败: {e}")

finally:
    # 关闭浏览器
    driver.quit()
